package leetcode_数组.二维数组及滚动数组;

import java.util.Arrays;

// 图片平滑器
public class _661 {
    public static void main(String[] args) {
        int[][] ints = {{2,3,4},{5,6,7},{8,9,10},{11,12,13},{14,15,16}};
        System.out.println(Arrays.deepToString(imageSmoother(ints)));
    }

    public static int[][] imageSmoother(int[][] M) {
        int m = M.length, n = M[0].length;
        int[][] ans = new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                ans[i][j] = M[i][j];
                int count = 1;
                if (i - 1 >= 0) { // 上方
                    ans[i][j] += M[i - 1][j];
                    count++;
                }
                if (i - 1 >= 0 && j - 1 >= 0) { // 左上方
                    ans[i][j] += M[i - 1][j - 1];
                    count++;
                }
                if (i - 1 >= 0 && j + 1 < n) { // 右上方
                    ans[i][j] += M[i - 1][j + 1];
                    count++;
                }
                if (j - 1 >= 0) { // 左方
                    ans[i][j] += M[i][j - 1];
                    count++;
                }
                if (j + 1 < n) { // 右方
                    ans[i][j] += M[i][j + 1];
                    count++;
                }
                if (i + 1 < m && j - 1 >= 0) { // 左下方
                    ans[i][j] += M[i + 1][j - 1];
                    count++;
                }
                if (i + 1 < m) { // 下方
                    ans[i][j] += M[i + 1][j];
                    count++;
                }
                if (i + 1 < m && j + 1 < n) { // 右下方
                    ans[i][j] += M[i + 1][j + 1];
                    count++;
                }
                ans[i][j] /= count;
            }
        }
        return ans;
    }

}
